from langchain.tools.render import format_tool_to_openai_function
from tools.tool_order_detail_query import search

from langchain.prompts import ChatPromptTemplate
from model.local_deep_seek_r1 import DeepSeekR1Prompt
 
#创建函数描述变量
functions = [
    format_tool_to_openai_function(f) for f in [
        search]
]
#定义llm
model = DeepSeekR1Prompt().bind(functions=functions)
 
#创建prompt
prompt = ChatPromptTemplate.from_messages([
    ("system", "你用于帮助查询数据"),
    ("user", "{input}"),
])
 
#定义chain
chain = prompt | model
 
#调用chain
chain.invoke({"input": "网关：云信留客山东移动行业,帮忙导出这个网关2月22日到24日的所有成功明细（发送内容，号码，时间，客户名称,状态，条数） @秦杨杨  "})